#pragma once
#include "RBTree.h"
namespace ape
{
	// set类
	template <class K>
	class set
	{
		// 仿函数operator() 获得key
		struct GetKeySet
		{
			const K &operator()(const K &key)
			{
				return key;
			}
		};

	private:
		RBTree<K, K, GetKeySet> _rbtree;

	public:
		/// ///////////////////  迭代器  /////////////////////////////////////

		typedef typename RBTree<K, K, GetKeySet>::const_itertaor iterator; // 普通迭代器也定义为const迭代器
		typedef typename RBTree<K, K, GetKeySet>::const_itertaor const_iterator;

		// begin迭代器
		iterator begin()
		{
			return _rbtree.begin();
		}
		// end迭代器
		iterator end()
		{
			return _rbtree.end();
		}
		// 插入
		pair<iterator, bool> insert(const K &key)
		{
			return _rbtree.Insert(key);
		}

		// 编译器自动生成默认构造函数
		set() = default;

		// 迭代器区间构造
		template <class InputIterator>
		set(InputIterator first, InputIterator last)
		{
			while (first != last)
			{
				_rbtree.Insert(*first);
				++first;
			}
		}

		iterator find(const K &k)
		{
			return _rbtree.Find(k);
		}
	};

	/////////////////// 测试 ////////////////////////////////////////

	void test_set()
	{
		// 插入
		int a[] = {16, 3, 7, 11, 9, 26, 18, 14, 15};
		set<int> s;
		for (auto e : a)
		{
			s.insert(e);
		}
		// 迭代器遍历
		set<int>::iterator it = s.begin();
		while (it != s.end())
		{
			cout << *it << " ";
			++it;
		}
		cout << endl;
		// for循环遍历
		for (auto e : s)
		{
			cout << e << " ";
		}
		cout << endl;
	}
}